#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
PDF批量加水印工具简易调用示例
"""

import os
from pdf_watermarker import process_folder

def main():
    """
    简单的示例函数，演示如何在代码中调用PDF批量水印功能
    """
    # 获取当前脚本所在文件夹作为输入文件夹
    current_folder = os.path.dirname(os.path.abspath(__file__))
    
    # 创建输出文件夹名称
    output_folder = os.path.join(current_folder, "水印处理结果")
    
    # 水印文件名
    watermark_filename = "watermark.pdf"
    
    print("=" * 50)
    print("PDF批量加水印工具示例")
    print("=" * 50)
    print(f"输入文件夹: {current_folder}")
    print(f"输出文件夹: {output_folder}")
    print(f"水印文件: {watermark_filename}")
    print("=" * 50)
    
    # 调用处理函数
    count = process_folder(current_folder, output_folder, watermark_filename)
    
    if count > 0:
        print(f"\n处理成功! 已处理 {count} 个PDF文件")
    else:
        print("\n处理失败! 请检查是否存在水印文件和PDF文件")

if __name__ == "__main__":
    main() 